#pragma once

#include <iostream>
#include <string>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <fcntl.h>
#include "../comm/util.hpp"
#include "../comm/log.hpp"

namespace ns_compiler
{
    using namespace ns_util;
    using namespace ns_log;
    class Compiler
    {
    public:
        Compiler() = default;
        ~Compiler() = default;

        // 返回值：编译成功——true；失败——false
        // 输入值：文件名——"xxx"
        // 代码文件："./temp/xxx.cpp"
        // 可执行文件："./temp/xxx.exe"
        // 错误文件："./temp/xxx.error"
        static bool Compile(const std::string& file_name)
        {
            pid_t pid = fork();
            // LOG(DEBUG) << ": " << PathUtil::Src(file_name) << std::endl;
            // LOG(DEBUG) << ": " << PathUtil::Exe(file_name) << std::endl;
            // LOG(DEBUG) << ": " << PathUtil::CompileError(file_name) << std::endl;
            
            if(pid < 0) // 创建子进程失败
            {
                LOG(ERROR) << "创建子进程失败" << std::endl;
                return false;
            }
            else if(pid == 0) // 子进程：完成编译的任务，并且编写错误日志
            {
                umask(0);
                int errfd = open(PathUtil::CompileError(file_name).c_str(), O_WRONLY | O_CREAT, 0644);
                if(errfd < 0)
                {
                    LOG(WARNING) << "创建错误文件失败" << std::endl;
                    exit(1);
                }

                // 将编译错误重定向到文件中
                dup2(errfd, 2);
                // 进程替换不会改变文件描述符，替换编译
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(), 
                PathUtil::Src(file_name).c_str(), "-std=c++11", "-D", "COMPILE_RUN", nullptr);
                // 走到这说明进程替换失败，报错
                LOG(ERROR) << "g++启动失败" << std::endl;
                exit(2);
            }
            else // 主进程：等待
            {
                pid_t child = waitpid(pid, nullptr, 0);
                if(child < 0)
                {
                    // 等待失败
                    LOG(ERROR) << "等待编译失败" << std::endl;
                    exit(3);
                }
                
                // 判断可执行文件是否存在
                if(FileUtil::isFileExist(PathUtil::Exe(file_name)))
                {
                   LOG(INFO) << PathUtil::Exe(file_name) << " 编译成功" << std::endl;
                   return true;
                }
            }
            LOG(WARNING) << "编译失败，未形成可执行文件" << std::endl;
            return false;
        }
    };
}